'''
加载json文件
'''
import json
from pathlib import Path
from pprint import pprint

from langchain_community.document_loaders import JSONLoader


# 加载json数据，返回json对象
def load_json_data(file_path):
    data = json.loads(Path(file_path).read_text())
    return data


def load_json_file():
    file_path = '../../load_file/simple_prompt.json'
    data = load_json_data(file_path)
    pprint(data)


# 提取 JSON 数据中 messages 键下的 content 字段的值
# 需要安装： pip install jq
def load_json_data_fromats(file_path):
    loader = JSONLoader(
        file_path=file_path,
        jq_schema='.messages[].content')
    data = loader.load()
    return data


# 加载json文件，并返回字符串
def load_json_file_fromat():
    file_path = '../../load_file/simple_prompt.json'
    data = load_json_data_fromats(file_path)
    pprint(data)
